437C - The Child and Toy - CodeForces Solution


graphs greedy sortings *1400

Please click on ads to support us..

Python Code:

n,m = map(int,input().split())
a = list(map(int,input().split()))
store = set()
b = list(range(1,n+1))

d = {}
for i in b:
    d[i] = [] 
    
for _ in range(m):
    x,y = map(int,input().split())
    store.add((x,y))
    store.add((y,x))
    d[x].append(y)
    d[y].append(x)
    
    
count = 0 
z = sorted(zip(a,b))

for _,i in z:
    
    for j in d[i]:
        
        if (i,j) not in store:
            continue 
            
        count+= a[i-1]
        store.remove((i,j))
        store.remove((j,i))
    if len(store)==0:
        break 
        
print(count)

C++ Code:

// author: Ali Safwat
#include <bits/extc++.h>
#define ordered_set tree<int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
#define Caraxes  ios_base :: sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
#define int long long int
#define F first
#define S second
using namespace std;
// using namespace __gnu_pbds;

template< typename  T = int >
std :: vector < T > get(int size = 0, std :: istream& in_stream = std :: cin);

const int N = 1e6+ 20;
void Dracarys(){
    int n,m;
    cin >> n >> m;
    map < int, int > mp;
    for(int i = 1;i <= n; ++i){
        int val;
        cin >> val;
        mp[i] = val;
    }
    int ans = 0;
    for(int i = 0;i < m; ++i){
        int u,v;
        cin >> u >> v;
        ans += (min(mp[u],mp[v]));
    }
    cout << ans << '\n';
}


signed main()
{
    Caraxes
    int test = 1;
    //cin >> test;
    while(test--) {
        Dracarys();
    }
}

template< typename  T>
std :: vector < T > get(int size , std :: istream& in_stream){
    if(!size){
        in_stream >> size;
    }
    std :: vector < T > array(size);
    for(auto& i : array)
        in_stream >> i;
    return array;
}


Comments

Submit
0 Comments
More Questions

1574A - Regular Bracket Sequences
1574B - Combinatorics Homework
1567A - Domino Disaster
1593A - Elections
1607A - Linear Keyboard
EQUALCOIN Equal Coins
XOREQN Xor Equation
MAKEPAL Weird Palindrome Making
HILLSEQ Hill Sequence
MAXBRIDGE Maximise the bridges
WLDRPL Wildcard Replacement
1221. Split a String in Balanced Strings
1002. Find Common Characters
1602A - Two Subsequences
1555A - PizzaForces
1607B - Odd Grasshopper
1084A - The Fair Nut and Elevator
1440B - Sum of Medians
1032A - Kitchen Utensils
1501B - Napoleon Cake
1584B - Coloring Rectangles
1562B - Scenes From a Memory
1521A - Nastia and Nearly Good Numbers
208. Implement Trie
1605B - Reverse Sort
1607C - Minimum Extraction
1604B - XOR Specia-LIS-t
1606B - Update Files
1598B - Groups
1602B - Divine Array